home *** CD-ROM | disk | FTP | other *** search
/ Personal Computer World 2005 October / PCWOCT05.iso / Software / FromTheMag / XAMPP 1.4.14 / xampp-win32-1.4.14-installer.exe / xampp / mysql / scripts / mysql_setpermission < prev    next >
Text File  |  2005-04-01  |  17KB  |  661 lines
  1. #!/usr/bin/perl
  2. ## Emacs, this is -*- perl -*- mode? :-)
  3. ##
  4. ##        Permission setter for MySQL
  5. ##
  6. ##        mady by Luuk de Boer (luuk@wxs.nl) 1998.
  7. ##        it's made under GPL ...:-))
  8. ##
  9. ##
  10. ############################################################################
  11. ## History
  12. ##
  13. ## 1.0 first start of the program
  14. ## 1.1 some changes from monty and after that
  15. ##     initial release in mysql 3.22.10 (nov 1998)
  16. ## 1.2 begin screen now in a loop + quit is using 0 instead of 9
  17. ##     after ideas of Paul DuBois.
  18. ## 1.2a Add Grant, References, Index and Alter privilege handling (Monty)
  19. ## 1.3 Applied patch provided by Martin Mokrejs <mmokrejs@natur.cuni.cz>
  20. ##     (General code cleanup, use the GRANT statement instead of updating
  21. ##     the privilege tables directly, added option to revoke privileges)
  22.  
  23. #### TODO
  24. #
  25. # empty ... suggestions ... mail them to me ...
  26.  
  27.  
  28. $version="1.3";
  29.  
  30. use DBI;
  31. use Getopt::Long;
  32. use strict;
  33. use vars qw($dbh $sth $hostname $opt_user $opt_password $opt_help $opt_host
  34.         $opt_socket $opt_port $host $version);
  35.  
  36. my $sqlhost = "";
  37. my $user = "";
  38.  
  39. $dbh=$host=$opt_user= $opt_password= $opt_help= $opt_host= $opt_socket= "";
  40. $opt_port=0;
  41.  
  42. read_my_cnf();        # Read options from ~/.my.cnf
  43.  
  44. GetOptions("user=s","password=s","help","host=s","socket=s","port=i");
  45.  
  46. usage() if ($opt_help); # the help function
  47.  
  48. if ($opt_host eq '')
  49. {
  50.   $sqlhost = "localhost";
  51. }
  52. else
  53. {
  54.   $sqlhost = $opt_host;
  55. }
  56.  
  57. # ask for a password if no password is set already
  58. if ($opt_password eq '')
  59. {
  60.   system "stty -echo";
  61.   print "Password for user $opt_user to connect to MySQL: ";
  62.   $opt_password = <STDIN>;
  63.   chomp($opt_password);
  64.   system "stty echo";
  65.   print "\n";
  66. }
  67.  
  68.  
  69. # make the connection to MySQL
  70. $dbh= DBI->connect("DBI:mysql:mysql:host=$sqlhost:port=$opt_port:mysql_socket=$opt_socket",$opt_user,$opt_password, {PrintError => 0}) ||
  71.   die("Can't make a connection to the mysql server.\n The error: $DBI::errstr");
  72.  
  73. # the start of the program
  74. &q1();
  75. exit(0); # the end...
  76.  
  77. #####
  78. # below all subroutines of the program
  79. #####
  80.  
  81. ###
  82. # the beginning of the program
  83. ###
  84. sub q1 { # first question ...
  85.   my ($answer,$end);
  86.   while (! $end) {
  87.     print "#"x70;
  88.     print "\n";
  89.     print "## Welcome to the permission setter $version for MySQL.\n";
  90.     print "## made by Luuk de Boer\n";
  91.     print "#"x70;
  92.     print "\n";
  93.     print "What would you like to do:\n";
  94.     print "  1. Set password for an existing user.\n";
  95.     print "  2. Create a database + user privilege for that database\n";
  96.         print "     and host combination (user can only do SELECT)\n";
  97.     print "  3. Create/append user privilege for an existing database\n";
  98.         print "     and host combination (user can only do SELECT)\n";
  99.     print "  4. Create/append broader user privileges for an existing\n";
  100.         print "     database and host combination\n";
  101.         print "     (user can do SELECT,INSERT,UPDATE,DELETE)\n";
  102.     print "  5. Create/append quite extended user privileges for an\n";
  103.         print "     existing database and host combination (user can do\n";
  104.         print "     SELECT,INSERT,UPDATE,DELETE,CREATE,DROP,INDEX,\n";
  105.         print "     LOCK TABLES,CREATE TEMPORARY TABLES)\n";
  106.     print "  6. Create/append database administrative privileges for an\n";
  107.         print "     existing database and host combination (user can do\n";
  108.         print "     SELECT,INSERT,UPDATE,DELETE,CREATE,DROP,INDEX,LOCK TABLES,\n";
  109.         print "     CREATE TEMPORARY TABLES,SHOW DATABASES,PROCESS)\n";
  110.     print "  7. Create/append full privileges for an existing database\n";
  111.         print "     and host combination (user has FULL privilege)\n";
  112.     print "  8. Remove all privileges for for an existing database and\n";
  113.         print "     host combination.\n";
  114.     print "     (user will have all permission fields set to N)\n";
  115.     print "  0. exit this program\n";
  116.     print "\nMake your choice [1,2,3,4,5,6,7,0]: ";
  117.     while (<STDIN>) {
  118.       $answer = $_;
  119.       chomp($answer);
  120.       if ($answer =~ /^[12345678]$/) {
  121.         if ($answer == 1) {
  122.        setpwd();
  123.         } elsif ($answer =~ /^[2345678]$/) {
  124.        addall($answer);
  125.     } else {
  126.           print "Sorry, something went wrong. With such option number you should not get here.\n\n";
  127.           $end = 1;
  128.         }
  129.       } elsif ($answer == 0) {
  130.         print "We hope we can help you next time \n\n";
  131.     $end = 1;
  132.       } else {
  133.         print "Your answer was $answer\n";
  134.         print "and that's wrong .... Try again\n";
  135.       }
  136.       last;
  137.     }
  138.   }
  139. }
  140.  
  141. ###
  142. # set a password for a user
  143. ###
  144. sub setpwd
  145. {
  146.   my ($user,$pass,$host) = "";
  147.   print "\n\nSetting a (new) password for a user.\n";
  148.  
  149.   $user = user();
  150.   $pass = newpass($user);
  151.   $host = hosts($user);
  152.  
  153.   print "#"x70;
  154.   print "\n\n";
  155.   print "That was it ... here is an overview of what you gave to me:\n";
  156.   print "The username         : $user\n";
  157. #  print "The password        : $pass\n";
  158.   print "The host        : $host\n";
  159.   print "#"x70;
  160.   print "\n\n";
  161.   print "Are you pretty sure you would like to implement this [yes/no]: ";
  162.   my $no = <STDIN>;
  163.   chomp($no);
  164.   if ($no =~ /n/i)
  165.   {
  166.     print "Okay .. that was it then ... See ya\n\n";
  167.     return(0);
  168.   }
  169.   else
  170.   {
  171.     print "Okay ... let's go then ...\n\n";
  172.   }
  173.   $user = $dbh->quote($user);
  174.   $host = $dbh->quote($host);
  175.   if ($pass eq '')
  176.   {
  177.     $pass = "''";
  178.   }
  179.   else
  180.   {
  181.     $pass = "PASSWORD(". $dbh->quote($pass) . ")";
  182.   }
  183.   my $sth = $dbh->prepare("update user set Password=$pass where User = $user and Host = $host") || die $dbh->errstr;
  184.   $sth->execute || die $dbh->errstr;
  185.   $sth->finish;
  186.   print "The password is set for user $user.\n\n";
  187.  
  188. }
  189.  
  190. ###
  191. # all things which will be added are done here
  192. ###
  193. sub addall {
  194.   my ($todo) = @_;
  195.   my ($answer,$good,$db,$user,$pass,$host,$priv);
  196.  
  197.   if ($todo == 2) {
  198.     $db = newdatabase();
  199.   } else {
  200.     $db = database();
  201.   }
  202.  
  203.   $user = newuser();
  204.   $pass = newpass("$user");
  205.   $host = newhosts();
  206.  
  207.   print "#"x70;
  208.   print "\n\n";
  209.   print "That was it ... here is an overview of what you gave to me:\n";
  210.   print "The database name    : $db\n";
  211.   print "The username         : $user\n";
  212. #  print "The password        : $pass\n";
  213.   print "The host(s)        : $host\n";
  214.   print "#"x70;
  215.   print "\n\n";
  216.   print "Are you pretty sure you would like to implement this [yes/no]: ";
  217.   my $no = <STDIN>;
  218.   chomp($no);
  219.   if ($no =~ /n/i) {
  220.     print "Okay .. that was it then ... See ya\n\n";
  221.     return(0);
  222.   } else {
  223.     print "Okay ... let's go then ...\n\n";
  224.   }
  225.  
  226.   if ($todo == 2) {
  227.     # create the database
  228.     if ($db) {
  229.     my $sth = $dbh->do("CREATE DATABASE $db") || $dbh->errstr;
  230.     } else {
  231.       print STDERR "What do you want? You wanted to create new database and add new user, right?\n";
  232.       die "But then specify databasename, please\n";
  233.   }
  234.   }
  235.  
  236.   if ( ( !$todo ) or not ( $todo =~ m/^[2-8]$/ ) ) {
  237.     print STDERR "Sorry, select option $todo isn't known inside the program .. See ya\n";
  238.     quit();
  239.   }
  240.  
  241.   my @hosts = split(/,/,$host);
  242.   if (!$user) {
  243.      die "username not specified: $user\n";
  244.   }
  245.   if (!$db) {
  246.      die "databasename is not specified nor *\n";
  247.   }
  248.   foreach $host (@hosts) {
  249.     # user privileges: SELECT
  250.     if (($todo == 2) || ($todo == 3)) {
  251.       $sth = $dbh->do("GRANT SELECT ON $db.* TO $user@\"$host\" IDENTIFIED BY \'$pass\'") || die $dbh->errstr;
  252.     } elsif ($todo == 4) {
  253.       # user privileges: SELECT,INSERT,UPDATE,DELETE
  254.       $sth = $dbh->do("GRANT SELECT,INSERT,UPDATE,DELETE ON $db.* TO $user@\"$host\" IDENTIFIED BY \'$pass\'") || die $dbh->errstr;
  255.     } elsif ($todo == 5) {
  256.       # user privileges: SELECT,INSERT,UPDATE,DELETE,CREATE,DROP,INDEX,LOCK TABLES,CREATE TEMPORARY TABLES
  257.       $sth = $dbh->do("GRANT SELECT,INSERT,UPDATE,DELETE,CREATE,DROP,INDEX,LOCK TABLES,CREATE TEMPORARY TABLES ON $db.* TO $user@\"$host\" IDENTIFIED BY \'$pass\'") || die $dbh->errstr;
  258.     } elsif ($todo == 6) {
  259.        # admin privileges: GRANT SELECT,INSERT,UPDATE,DELETE,CREATE,DROP,INDEX,LOCK TABLES,CREATE TEMPORARY TABLES,SHOW DATABASES,PROCESS
  260.        $sth = $dbh->do("GRANT SELECT,INSERT,UPDATE,DELETE,CREATE,DROP,INDEX,LOCK TABLES,CREATE TEMPORARY TABLES,SHOW DATABASES,PROCESS ON $db.* TO $user@\"$host\" IDENTIFIED BY \'$pass\'") || die $dbh->errstr;
  261.     } elsif ($todo == 7) {
  262.        # all privileges
  263.        $sth = $dbh->do("GRANT ALL ON $db.* TO \'$user\'\@\'$host\' IDENTIFIED BY \'$pass\'") || die $dbh->errstr;
  264.     } elsif ($todo == 8) {
  265.        # all privileges set to N
  266.        $sth = $dbh->do("REVOKE ALL ON *.* FROM \'$user\'\@\'$host\'") || die $dbh->errstr;
  267.     }
  268.     }
  269.   $dbh->do("FLUSH PRIVILEGES") || print STDERR "Can't flush privileges\n";
  270.   print "Everything is inserted and mysql privileges have been reloaded.\n\n";
  271. }
  272.  
  273. ###
  274. # ask for a new database name
  275. ###
  276. sub newdatabase {
  277.   my ($answer,$good,$db);
  278.   print "\n\nWhich database would you like to add: ";
  279.   while (<STDIN>) {
  280.     $answer = $_;
  281.     $good = 0;
  282.     chomp($answer);
  283.     if ($answer) {
  284.       my $sth = $dbh->prepare("SHOW DATABASES") || die $dbh->errstr;
  285.       $sth->execute || die $dbh->errstr;
  286.       while (my @r = $sth->fetchrow_array) {
  287.         if ($r[0] eq $answer) {
  288.           print "\n\nSorry, this database name is already in use; try something else: ";
  289.           $good = 1;
  290.         }
  291.       }
  292.     } else {
  293.       print "You must type something ...\nTry again: ";
  294.       next;
  295.     }
  296.     last if ($good == 0);
  297.   }
  298.   $db = $answer;
  299.   print "The new database $db will be created\n";
  300.   return($db);
  301. }
  302.  
  303. ###
  304. # select a database
  305. ###
  306. sub database {
  307.   my ($answer,$good,$db);
  308.   print "\n\nWhich database from existing databases would you like to select: \n";
  309.   print "You can choose from: \n";
  310.   my $sth = $dbh->prepare("show databases") || die $dbh->errstr;
  311.   $sth->execute || die $dbh->errstr;
  312.   while (my @r = $sth->fetchrow_array) {
  313.     print "  - $r[0] \n";
  314.   }
  315.   print "Which database will it be (case sensitive). Type * for any: \n";
  316.   while (<STDIN>) {
  317.     $answer = $_;
  318.     $good = 0;
  319.     chomp($answer);
  320.     if ($answer) {
  321.       if ($answer eq "*") {
  322.         print "OK, the user entry will NOT be limited to any database";
  323.         return("*");
  324.       }
  325.       my $sth = $dbh->prepare("show databases") || die $dbh->errstr;
  326.       $sth->execute || die $dbh->errstr;
  327.       while (my @r = $sth->fetchrow_array) {
  328.         if ($r[0] eq $answer) {
  329.           $good = 1;
  330.           $db = $r[0];
  331.           last;
  332.         }
  333.       }
  334.     } else {
  335.         print "Type either database name or * meaning any databasename. That means";
  336.         print " any of those above but also any which will be created in future!";
  337.         print " This option gives a user chance to operate on databse mysql, which";
  338.         print " contains privilege settings. That is really risky!\n";
  339.       next;
  340.     }
  341.     if ($good == 1) {
  342.       last;
  343.     } else {
  344.       print "You must select one from the list.\nTry again: ";
  345.       next;
  346.     }
  347.   }
  348.   print "The database $db will be used.\n";
  349.   return($db);
  350. }
  351.  
  352. ###
  353. # ask for a new username
  354. ###
  355. sub newuser
  356. {
  357.   my $user = "";
  358.   my $answer = "";
  359.  
  360.   print "\nWhat username is to be created: ";
  361.   while(<STDIN>)
  362.   {
  363.     $answer = $_;
  364.     chomp($answer);
  365.     if ($answer)
  366.     {
  367.       $user = $answer;
  368.     }
  369.     else
  370.     {
  371.       print "You must type something ...\nTry again: ";
  372.       next;
  373.     }
  374.     last;
  375.   }
  376.   print "Username = $user\n";
  377.   return($user);
  378. }
  379.  
  380. ###
  381. # ask for a user which is already in the user table
  382. ###
  383. sub user
  384. {
  385.   my ($answer,$user);
  386.  
  387.   print "\nFor which user do you want to specify a password: ";
  388.   while(<STDIN>)
  389.   {
  390.     $answer = $_;
  391.     chomp($answer);
  392.     if ($answer)
  393.     {
  394.       my $sth = $dbh->prepare("select User from user where User = '$answer'") || die $dbh->errstr;
  395.       $sth->execute || die $dbh->errstr;
  396.       my @r = $sth->fetchrow_array;
  397.       if ($r[0])
  398.       {
  399.         $user = $r[0];
  400.       }
  401.       else
  402.       {
  403.        print "Sorry, user $answer isn't known in the user table.\nTry again: ";
  404.        next;
  405.      }
  406.     }
  407.     else
  408.     {
  409.       print "You must type something ...\nTry again: ";
  410.       next;
  411.     }
  412.     last;
  413.   }
  414.   print "Username = $user\n";
  415.   return($user);
  416. }
  417.  
  418. ###
  419. # ask for a new password
  420. ###
  421. sub newpass
  422. {
  423.   my ($user) = @_;
  424.   my ($pass,$answer,$good,$yes);
  425.  
  426.   print "Would you like to set a password for $user [y/n]: ";
  427.   $yes = <STDIN>;
  428.   chomp($yes);
  429.   if ($yes =~ /y/)
  430.   {
  431.     system "stty -echo";
  432.     print "What password do you want to specify for $user: ";
  433.     while(<STDIN>)
  434.     {
  435.       $answer = $_;
  436.       chomp($answer);
  437.       system "stty echo";
  438.       print "\n";
  439.       if ($answer)
  440.       {
  441.         system "stty -echo";
  442.         print "Type the password again: ";
  443.         my $second = <STDIN>;
  444.         chomp($second);
  445.         system "stty echo";
  446.         print "\n";
  447.         if ($answer ne $second)
  448.         {
  449.           print "Passwords aren't the same; we begin from scratch again.\n";
  450.           system "stty -echo";
  451.           print "Password please: ";
  452.           next;
  453.         }
  454.         else
  455.         {
  456.           $pass = $answer;
  457.         }
  458.       }
  459.       else
  460.       {
  461.         print "You must type something ...\nTry again: ";
  462.         next;
  463.       }
  464.       last;
  465.     }
  466. #    print "The password for $user is $pass.\n";
  467.   }
  468.   else
  469.   {
  470.     print "We won't set a password so the user doesn't have to use it\n";
  471.     $pass = "";
  472.   }
  473.   return($pass);
  474. }
  475.  
  476. ###
  477. # ask for new hosts
  478. ###
  479. sub newhosts
  480. {
  481.   my ($host,$answer,$good);
  482.  
  483.   print "We now need to know from what host(s) the user will connect.\n";
  484.   print "Keep in mind that % means 'from any host' ...\n";
  485.   print "The host please: ";
  486.   while(<STDIN>)
  487.   {
  488.     $answer = $_;
  489.     chomp($answer);
  490.     if ($answer)
  491.     {
  492.       $host .= ",$answer";
  493.       print "Would you like to add another host [yes/no]: ";
  494.       my $yes = <STDIN>;
  495.       chomp($yes);
  496.       if ($yes =~ /y/i)
  497.       {
  498.         print "Okay, give us the host please: ";
  499.         next;
  500.       }
  501.       else
  502.       {
  503.         print "Okay we keep it with this ...\n";
  504.       }
  505.     }
  506.     else
  507.     {
  508.       print "You must type something ...\nTry again: ";
  509.       next;
  510.     }
  511.     last;
  512.   }
  513.   $host =~ s/^,//;
  514.   print "The following host(s) will be used: $host.\n";
  515.   return($host);
  516. }
  517.  
  518. ###
  519. # ask for a host which is already in the user table
  520. ###
  521. sub hosts
  522. {
  523.   my ($user) = @_;
  524.   my ($answer,$good,$host);
  525.  
  526.   print "We now need to know which host for $user we have to change.\n";
  527.   print "Choose from the following hosts: \n";
  528.   $user = $dbh->quote($user);
  529.   my $sth = $dbh->prepare("select Host,User from user where User = $user") || die $dbh->errstr;
  530.   $sth->execute || die $dbh->errstr;
  531.   while (my @r = $sth->fetchrow_array)
  532.   {
  533.     print "  - $r[0] \n";
  534.   }
  535.   print "The host please (case sensitive): ";
  536.   while(<STDIN>)
  537.   {
  538.     $answer = $_;
  539.     chomp($answer);
  540.     if ($answer)
  541.     {
  542.       $sth = $dbh->prepare("select Host,User from user where Host = '$answer' and User = $user") || die $dbh->errstr;
  543.       $sth->execute || die $dbh->errstr;
  544.       my @r = $sth->fetchrow_array;
  545.       if ($r[0])
  546.       {
  547.         $host = $answer;
  548.         last;
  549.       }
  550.       else
  551.       {
  552.         print "You have to select a host from the list ...\nTry again: ";
  553.         next;
  554.       }
  555.     }
  556.     else
  557.     {
  558.       print "You have to type something ...\nTry again: ";
  559.       next;
  560.     }
  561.     last;
  562.   }
  563.   print "The following host will be used: $host.\n";
  564.   return($host);
  565. }
  566.  
  567. ###
  568. # a nice quit (first disconnect and then exit
  569. ###
  570. sub quit
  571. {
  572.   $dbh->disconnect;
  573.   exit(0);
  574. }
  575.  
  576. ###
  577. # Read variables password, port and socket from .my.cnf under the client
  578. # or perl groups
  579. ###
  580.  
  581. sub read_my_cnf
  582. {
  583.   open(TMP,$ENV{'HOME'} . "/.my.cnf") || return 1;
  584.   while (<TMP>)
  585.   {
  586.     if (/^\[(client|perl)\]/i)
  587.     {
  588.       while ((defined($_=<TMP>)) && !/^\[\w+\]/)
  589.       {
  590.     print $_;
  591.     if (/^host\s*=\s*(\S+)/i)
  592.     {
  593.       $opt_host = $1;
  594.     }
  595.     elsif (/^user\s*=\s*(\S+)/i)
  596.     {
  597.       $opt_user = $1;
  598.     }
  599.     elsif (/^password\s*=\s*(\S+)/i)
  600.     {
  601.       $opt_password = $1;
  602.     }
  603.     elsif (/^port\s*=\s*(\S+)/i)
  604.     {
  605.       $opt_port = $1;
  606.     }
  607.     elsif (/^socket\s*=\s*(\S+)/i)
  608.     {
  609.       $opt_socket = $1;
  610.     }
  611.       }
  612.     }
  613.   }
  614.   close(TMP);
  615. }
  616.  
  617. ###
  618. # the help text
  619. ###
  620. sub usage
  621. {
  622.   print <<EOL;
  623. ----------------------------------------------------------------------
  624.                  The permission setter for MySQL.
  625.                       version: $version
  626.  
  627.                  made by: Luuk de Boer <luuk\@wxs.nl>
  628. ----------------------------------------------------------------------
  629.  
  630. The permission setter is a little program which can help you add users
  631. or databases or change passwords in MySQL. Keep in mind that we don't
  632. check permissions which already been set in MySQL. So if you can't
  633. connect to MySQL using the permission you just added, take a look at
  634. the permissions which have already been set in MySQL.
  635.  
  636. The permission setter first reads your .my.cnf file in your Home
  637. directory if it exists.
  638.  
  639. Options for the permission setter:
  640.  
  641. --help        : print this help message and exit.
  642.  
  643. The options shown below are used for making the connection to the MySQL
  644. server. Keep in mind that the permissions for the user specified via
  645. these options must be sufficient to add users / create databases / set
  646. passwords.
  647.  
  648. --user        : is the username to connect with.
  649. --password    : the password of the username.
  650. --host        : the host to connect to.
  651. --socket    : the socket to connect to.
  652. --port        : the port number of the host to connect to.
  653.  
  654. If you don't give a password and no password is set in your .my.cnf
  655. file, then the permission setter will ask for a password.
  656.  
  657.  
  658. EOL
  659. exit(0);
  660. }
  661.